home *** CD-ROM | disk | FTP | other *** search
/ Night Owl 7 / Night Owl Shareware (NOPV7)(Night Owl Publisher Inc.)(1992).bin / 038a / bash1_12.arj / BASH1-12.TAR / bash-1.12 / lib / glob / glob.c < prev    next >
C/C++ Source or Header  |  1992-01-04  |  12KB  |  514 lines
  1. /* File-name wildcard pattern matching for GNU.
  2.    Copyright (C) 1985, 1988, 1989 Free Software Foundation, Inc.
  3.  
  4.    This program is free software; you can redistribute it and/or modify
  5.    it under the terms of the GNU General Public License as published by
  6.    the Free Software Foundation; either version 1, or (at your option)
  7.    any later version.
  8.  
  9.    This program is distributed in the hope that it will be useful,
  10.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.    GNU General Public License for more details.
  13.  
  14.    You should have received a copy of the GNU General Public License
  15.    along with this program; if not, write to the Free Software
  16.    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */
  17.  
  18. /* To whomever it may concern: I have never seen the code which most
  19.    Unix programs use to perform this function.  I wrote this from scratch
  20.    based on specifications for the pattern matching.  --RMS.  */
  21.  
  22. #if defined (SHELL)
  23. #  include <config.h>
  24. #endif
  25.  
  26. #if defined (USG) && !defined (Xenix)
  27. #  if !defined (USGr3)
  28. #    define USGr3
  29. #endif /* USGr3 */
  30. #endif /* USG && !Xenix */
  31.  
  32. #include <sys/types.h>
  33.  
  34. #if defined (_POSIX_VERSION) || defined (USGr3) || defined (DIRENT)
  35. #  include <dirent.h>
  36. #  if !defined (direct)
  37. #    define direct dirent
  38. #  endif /* !direct */
  39. #  define       D_NAMLEN(d) strlen((d)->d_name)
  40. #else
  41. #  define D_NAMLEN(d) ((d)->d_namlen)
  42. #  if defined (Xenix)
  43. #    include <sys/ndir.h>
  44. #  else
  45. #    if defined (USG)
  46. #      include "ndir.h"
  47. #     else
  48. #      include <sys/dir.h>
  49. #    endif
  50. #  endif
  51. #endif    /* USGr3 || DIRENT.  */
  52.  
  53. #if defined (_POSIX_SOURCE)
  54. /* Posix does not require that the d_ino field be present, and some
  55.    systems do not provide it. */
  56. #  define REAL_DIR_ENTRY(dp) 1
  57. #else
  58. #  define REAL_DIR_ENTRY(dp) (dp->d_ino != 0)
  59. #endif /* _POSIX_SOURCE */
  60.  
  61. #if defined (NeXT)
  62. #include <string.h>
  63. #else
  64. #if defined (USG)
  65. #if !defined (isc386)
  66. #  include <memory.h>
  67. #endif
  68. #include <string.h>
  69. #if defined (RISC6000)
  70. extern void bcopy ();
  71. #else /* RISC6000 */
  72. #define bcopy(s, d, n) ((void) memcpy ((d), (s), (n)))
  73. #endif /* RISC6000 */
  74. #define rindex    strrchr
  75.  
  76. #else /* !USG */
  77. #include <strings.h>
  78. #endif /* !USG */
  79. #endif /* !NeXT */
  80.  
  81. #include "fnmatch.h"
  82.  
  83. /* If the opendir () on your system lets you open non-directory files,
  84.    then we consider that not robust.  Define OPENDIR_NOT_ROBUST in the
  85.    SYSDEP_CFLAGS for your machines entry in machines.h. */
  86. #if defined (OPENDIR_NOT_ROBUST)
  87. #if defined (SHELL)
  88. # include "posixstat.h"
  89. #else
  90. # include <sys/stat.h>
  91. #endif /* SHELL */
  92. #endif /* OPENDIR_NOT_ROBUST */
  93.  
  94. extern char *malloc (), *realloc ();
  95. extern void free ();
  96.  
  97. #ifndef NULL
  98. #define NULL 0
  99. #endif
  100.  
  101. /* Global variable which controls whether or not * matches .*.
  102.    Non-zero means don't match .*.  */
  103. int noglob_dot_filenames = 1;
  104.  
  105.  
  106. /* Return nonzero if PATTERN has any special globbing chars in it.  */
  107. int
  108. glob_pattern_p (pattern)
  109.      char *pattern;
  110. {
  111.   register char *p = pattern;
  112.   register char c;
  113.   int    open = 0;
  114.  
  115.   while ((c = *p++) != '\0')
  116.     switch (c)
  117.       {
  118.       case '?':
  119.       case '*':
  120.     return (1);
  121.  
  122.       case '[':        /* Only accept an open brace if there is a close */
  123.     open++;        /* brace to match it.  Bracket expressions must be */
  124.     continue;    /* complete, according to Posix.2 */
  125.       case ']':
  126.     if (open)
  127.       return (1);
  128.     continue;      
  129.  
  130.       case '\\':
  131.     if (*p++ == '\0')
  132.       return (0);
  133.       }
  134.  
  135.   return (0);
  136. }
  137.  
  138. /* Return a vector of names of files in directory DIR
  139.    whose names match glob pattern PAT.
  140.    The names are not in any particular order.
  141.    Wildcards at the beginning of PAT do not match an initial period.
  142.  
  143.    The vector is terminated by an element that is a null pointer.
  144.  
  145.    To free the space allocated, first free the vector's elements,
  146.    then free the vector.
  147.  
  148.    Return 0 if cannot get enough memory to hold the pointer
  149.    and the names.
  150.  
  151.    Return -1 if cannot access directory DIR.
  152.    Look in errno for more information.  */
  153.  
  154. char **
  155. glob_vector (pat, dir)
  156.      char *pat;
  157.      char *dir;
  158. {
  159.   struct globval
  160.     {
  161.       struct globval *next;
  162.       char *name;
  163.     };
  164.  
  165.   DIR *d;
  166.   register struct direct *dp;
  167.   struct globval *lastlink;
  168.   register struct globval *nextlink;
  169.   register char *nextname;
  170.   unsigned int count;
  171.   int lose;
  172.   register char **name_vector;
  173.   register unsigned int i;
  174. #if defined (OPENDIR_NOT_ROBUST)
  175.   struct stat finfo;
  176.  
  177.   if (stat (dir, &finfo) < 0)
  178.     return ((char **) -1);
  179.  
  180.   if (!S_ISDIR (finfo.st_mode))
  181.     return ((char **) -1);
  182. #endif /* OPENDIR_NOT_ROBUST */
  183.  
  184.   d = opendir (dir);
  185.   if (d == NULL)
  186.     return ((char **) -1);
  187.  
  188.   lastlink = 0;
  189.   count = 0;
  190.   lose = 0;
  191.  
  192.   /* Scan the directory, finding all names that match.
  193.      For each name that matches, allocate a struct globval
  194.      on the stack and store the name in it.
  195.      Chain those structs together; lastlink is the front of the chain.  */
  196.   while (1)
  197.     {
  198.       int flags;        /* Flags passed to fnmatch (). */
  199. #if defined (SHELL)
  200.       /* Make globbing interruptible in the bash shell. */
  201.       extern int interrupt_state;
  202.  
  203.       if (interrupt_state)
  204.     {
  205.       closedir (d);
  206.       lose = 1;
  207.       goto lost;
  208.     }
  209. #endif /* SHELL */
  210.       
  211.       dp = readdir (d);
  212.       if (dp == NULL)
  213.     break;
  214.  
  215.       /* If this directory entry is not to be used, try again. */
  216.       if (!REAL_DIR_ENTRY (dp))
  217.     continue;
  218.  
  219.       /* If a dot must be explicity matched, check to see if they do. */
  220.       if (noglob_dot_filenames && dp->d_name[0] == '.' && pat[0] != '.')
  221.     continue;
  222.  
  223.       flags = (noglob_dot_filenames ? FNM_PERIOD : 0) | FNM_PATHNAME;
  224.  
  225.       if (fnmatch (pat, dp->d_name, flags) != FNM_NOMATCH)
  226.     {
  227.       nextlink = (struct globval *) alloca (sizeof (struct globval));
  228.       nextlink->next = lastlink;
  229.       nextname = (char *) malloc (D_NAMLEN (dp) + 1);
  230.       if (nextname == NULL)
  231.         {
  232.           lose = 1;
  233.           break;
  234.         }
  235.       lastlink = nextlink;
  236.       nextlink->name = nextname;
  237.       bcopy (dp->d_name, nextname, D_NAMLEN (dp) + 1);
  238.       ++count;
  239.     }
  240.     }
  241.   (void) closedir (d);
  242.  
  243.   if (!lose)
  244.     {
  245.       name_vector = (char **) malloc ((count + 1) * sizeof (char *));
  246.       lose |= name_vector == NULL;
  247.     }
  248.  
  249.   /* Have we run out of memory?     */
  250.  lost:
  251.   if (lose)
  252.     {
  253.       /* Here free the strings we have got.  */
  254.       while (lastlink)
  255.     {
  256.       free (lastlink->name);
  257.       lastlink = lastlink->next;
  258.     }
  259.       return (NULL);
  260.     }
  261.  
  262.   /* Copy the name pointers from the linked list into the vector.  */
  263.   for (i = 0; i < count; ++i)
  264.     {
  265.       name_vector[i] = lastlink->name;
  266.       lastlink = lastlink->next;
  267.     }
  268.  
  269.   name_vector[count] = NULL;
  270.   return (name_vector);
  271. }
  272.  
  273. /* Return a new array which is the concatenation
  274.    of each string in ARRAY to DIR. */
  275.  
  276. static char **
  277. glob_dir_to_array (dir, array)
  278.      char *dir, **array;
  279. {
  280.   register unsigned int i, l;
  281.   int add_slash;
  282.   char **result;
  283.  
  284.   l = strlen (dir);
  285.   if (l == 0)
  286.     return (array);
  287.  
  288.   add_slash = dir[l - 1] != '/';
  289.  
  290.   i = 0;
  291.   while (array[i] != NULL)
  292.     ++i;
  293.  
  294.   result = (char **) malloc ((i + 1) * sizeof (char *));
  295.   if (result == NULL)
  296.     return (NULL);
  297.  
  298.   for (i = 0; array[i] != NULL; i++)
  299.     {
  300.       result[i] = (char *) malloc (l + (add_slash ? 1 : 0)
  301.                    + strlen (array[i]) + 1);
  302.       if (result[i] == NULL)
  303.     return (NULL);
  304.       sprintf (result[i], "%s%s%s", dir, add_slash ? "/" : "", array[i]);
  305.     }
  306.   result[i] = NULL;
  307.  
  308.   /* Free the input array.  */
  309.   for (i = 0; array[i] != NULL; i++)
  310.     free (array[i]);
  311.   free ((char *) array);
  312.  
  313.   return (result);
  314. }
  315.  
  316. /* Do globbing on PATHNAME.  Return an array of pathnames that match,
  317.    marking the end of the array with a null-pointer as an element.
  318.    If no pathnames match, then the array is empty (first element is null).
  319.    If there isn't enough memory, then return NULL.
  320.    If a file system error occurs, return -1; `errno' has the error code.  */
  321. char **
  322. glob_filename (pathname)
  323.      char *pathname;
  324. {
  325.   char **result;
  326.   unsigned int result_size;
  327.   char *directory_name, *filename;
  328.   unsigned int directory_len;
  329.  
  330.   result = (char **) malloc (sizeof (char *));
  331.   result_size = 1;
  332.   if (result == NULL)
  333.     return (NULL);
  334.  
  335.   result[0] = NULL;
  336.  
  337.   /* Find the filename.  */
  338.   filename = rindex (pathname, '/');
  339.   if (filename == NULL)
  340.     {
  341.       filename = pathname;
  342.       directory_name = "";
  343.       directory_len = 0;
  344.     }
  345.   else
  346.     {
  347.       directory_len = (filename - pathname) + 1;
  348.       directory_name = (char *) alloca (directory_len + 1);
  349.  
  350.       bcopy (pathname, directory_name, directory_len);
  351.       directory_name[directory_len] = '\0';
  352.       ++filename;
  353.     }
  354.  
  355.   /* If directory_name contains globbing characters, then we
  356.      have to expand the previous levels.  Just recurse. */
  357.   if (glob_pattern_p (directory_name))
  358.     {
  359.       char **directories;
  360.       register unsigned int i;
  361.  
  362.       if (directory_name[directory_len - 1] == '/')
  363.     directory_name[directory_len - 1] = '\0';
  364.  
  365.       directories = glob_filename (directory_name);
  366.  
  367.       if (directories == NULL)
  368.     goto memory_error;
  369.       else if ((int) directories == -1)
  370.     return ((char **) -1);
  371.       else if (*directories == NULL)
  372.     {
  373.       free ((char *) directories);
  374.       return ((char **) -1);
  375.     }
  376.  
  377.       /* We have successfully globbed the preceding directory name.
  378.      For each name in DIRECTORIES, call glob_vector on it and
  379.      FILENAME.  Concatenate the results together.  */
  380.       for (i = 0; directories[i] != NULL; ++i)
  381.     {
  382.       char **temp_results;
  383.  
  384.       if (filename && filename[0])
  385.         temp_results = glob_vector (filename, directories[i]);
  386.       else
  387.         {
  388.           /* Null pathname, do not scan the directory. */
  389.           temp_results = (char **)malloc (2 * sizeof (char *));
  390.  
  391.           if (!temp_results)
  392.         goto memory_error;
  393.  
  394.           temp_results[0] = (char *)malloc (1);
  395.  
  396.           if (!temp_results[0])
  397.         goto memory_error;
  398.  
  399.           temp_results[0][0] = '\0';
  400.           temp_results[1] = (char *)NULL;
  401.         }
  402.  
  403.       /* Handle error cases. */
  404.       if (temp_results == NULL)
  405.         goto memory_error;
  406.       else if (temp_results == (char **)-1)
  407.         /* This filename is probably not a directory.  Ignore it.  */
  408.         ;
  409.       else
  410.         {
  411.           char **array;
  412.           register unsigned int l;
  413.  
  414.           array = glob_dir_to_array (directories[i], temp_results);
  415.           l = 0;
  416.           while (array[l] != NULL)
  417.         ++l;
  418.  
  419.           result =
  420.         (char **)realloc (result, (result_size + l) * sizeof (char *));
  421.  
  422.           if (result == NULL)
  423.         goto memory_error;
  424.  
  425.           for (l = 0; array[l] != NULL; ++l)
  426.         result[result_size++ - 1] = array[l];
  427.  
  428.           result[result_size - 1] = NULL;
  429.  
  430.           /* Note that the elements of ARRAY are not freed.  */
  431.           free ((char *) array);
  432.         }
  433.     }
  434.       /* Free the directories.  */
  435.       for (i = 0; directories[i]; i++)
  436.     free (directories[i]);
  437.  
  438.       free ((char *) directories);
  439.  
  440.       return (result);
  441.     }
  442.  
  443.   /* If there is only a directory name, return it. */
  444.   if (*filename == '\0')
  445.     {
  446.       result = (char **) realloc ((char *) result, 2 * sizeof (char *));
  447.       if (result == NULL)
  448.     return (NULL);
  449.       result[0] = (char *) malloc (directory_len + 1);
  450.       if (result[0] == NULL)
  451.     goto memory_error;
  452.       bcopy (directory_name, result[0], directory_len + 1);
  453.       result[1] = NULL;
  454.       return (result);
  455.     }
  456.   else
  457.     {
  458.       /* Otherwise, just return what glob_vector
  459.      returns appended to the directory name. */
  460.       char **temp_results = glob_vector (filename,
  461.                      (directory_len == 0
  462.                       ? "." : directory_name));
  463.  
  464.       if (temp_results == NULL || temp_results == (char **)-1)
  465.     return (temp_results);
  466.  
  467.       return (glob_dir_to_array (directory_name, temp_results));
  468.     }
  469.  
  470.   /* We get to memory error if the program has run out of memory, or
  471.      if this is the shell, and we have been interrupted. */
  472.  memory_error:
  473.   if (result != NULL)
  474.     {
  475.       register unsigned int i;
  476.       for (i = 0; result[i] != NULL; ++i)
  477.     free (result[i]);
  478.       free ((char *) result);
  479.     }
  480. #if defined (SHELL)
  481.   {
  482.     extern int interrupt_state;
  483.  
  484.     if (interrupt_state)
  485.       throw_to_top_level ();
  486.   }
  487. #endif /* SHELL */
  488.   return (NULL);
  489. }
  490.  
  491. #ifdef TEST
  492.  
  493. main (argc, argv)
  494.      int argc;
  495.      char **argv;
  496. {
  497.   unsigned int i;
  498.  
  499.   for (i = 1; i < argc; ++i)
  500.     {
  501.       char **value = glob_filename (argv[i]);
  502.       if (value == NULL)
  503.     puts ("Out of memory.");
  504.       else if ((int) value == -1)
  505.     perror (argv[i]);
  506.       else
  507.     for (i = 0; value[i] != NULL; i++)
  508.       puts (value[i]);
  509.     }
  510.  
  511.   exit (0);
  512. }
  513. #endif    /* TEST.  */
  514.